from BaseClassfyModels import BaseClassfyModels
from startAndShow import showAndRecoedScores

class MyClassfyModels(showAndRecoedScores,BaseClassfyModels):
    def __init__(self, X_train, y_train, X_test, y_test):
        BaseClassfyModels.__init__(self, X_train, y_train, X_test, y_test)

if __name__ == '__main__':
    import pandas as pd
    from sklearn.model_selection import train_test_split

    data = pd.read_excel(r'./robotStateSimpleTest.xlsx')
    features = data.iloc[:, :-1]
    lable = data.iloc[:, -1:]  # 最后一个是标签
    X_train, X_test, y_train, y_test = train_test_split(features, lable, train_size=0.8, random_state=8)
    mymodels=MyClassfyModels(X_train=X_train,X_test=X_test,y_train=y_train.values.ravel(),y_test=y_test.values.ravel())
    mymodels.startAllAndShowResult()